package main

import (
	"bufio"
	"fmt"
	"io/ioutil"
	"os"
	"io"
)

//错误检查
func check(e error) {
	if e != nil {
		panic(e)
	}
}

//读写文件
func main() {

	//写文件
	//通过ioutil写入文件
	d1 := []byte("hello\ngo\n")
	err := ioutil.WriteFile("/tmp/dat", d1, 0644)
	check(err)

	//更细粒度的写入，先打开一个文件
	f, err := os.Create("/tmp/dat1")
	check(err)
	//然后习惯立即使用 defer 调用文件的 Close操作
	defer f.Close()
	//后写入内容，Write写入字节，WriteString写入字符串
	d2 := []byte{115, 111, 109, 101, 10}
	n2, err := f.Write(d2)
	check(err)
	fmt.Printf("wrote %d bytes\n", n2)
	n3, err := f.WriteString("writes\n")
	check(err)
	fmt.Printf("wrote %d bytes\n", n3)
	//后调用 Sync来将缓冲区的信息写入磁盘
	f.Sync()

	//bufio 提供了和前面的带缓冲的读取器一样的带缓冲的写入器。
	w := bufio.NewWriter(f)
	n4, err := w.WriteString("buffered\n")
	check(err)
	fmt.Printf("wrote %d bytes\n", n4)
	//使用 Flush 来确保所有缓存的操作已写入底层写入器
	w.Flush()

	//读文件
	//通过ioutil读文件
	dat, err := ioutil.ReadFile("/tmp/dat")
	check(err)
	fmt.Print(string(dat))

	//从使用 os.Open打开一个文件
	f1, err := os.Open("/tmp/dat")
	check(err)
	defer f1.Close()
	//从文件开始位置读取一些字节。这里最多读取 5 个字节
	b1 := make([]byte, 5)
	m1, err := f1.Read(b1)
	check(err)
	fmt.Printf("%d bytes: %s\n", m1, string(b1))
	//也可以 Seek 到一个文件中已知的位置并从这个位置开始进行读取
	o2, err := f1.Seek(6, 0)
	check(err)
	b2 := make([]byte, 2)
	m2, err := f1.Read(b2)
	check(err)
	fmt.Printf("%d bytes @ %d: %s\n", m2, o2, string(b2))
	//io 包提供了一些可以帮助我们进行文件读取的函数
	//上面的读取可以使用 ReadAtLeast 得到一个更健壮的实现
	o3, err := f1.Seek(6, 0)
	check(err)
	b3 := make([]byte, 2)
	m3, err := io.ReadAtLeast(f1, b3, 2)
	check(err)
	fmt.Printf("%d bytes @ %d: %s\n", m3, o3, string(b3))
	//没有内置的回转支持，但是使用 Seek(0, 0) 实现
	_, err = f1.Seek(0, 0)
	check(err)

	//bufio 包实现了带缓冲的读取，这不仅对有很多小的读取操作的能提升性能，也提供了很多附加的读取函数
	r4 := bufio.NewReader(f1)
	b4, err := r4.Peek(5)
	check(err)
	fmt.Printf("5 bytes: %s\n", string(b4))

}
